import requests
from threading import Thread
import os
import time
import sys
import random
'''
单线程
更加严密的逻辑判断
'''


def download(url, saveUrl):
    zip = url.split('_')[-1]
    lat_lon = zip.split('.')[0]
    randomUserAgent = getRandomUserAgent()
    headers = {'User-Agent': randomUserAgent}
    try:
        res = requests.get(url, headers=headers, timeout=(30, 30), stream=True)
        res_code = res.status_code
        res_size = int(res.headers['Content-Length'])

        if(res_code == 200):
            log('[200]__%s__Start\n' % lat_lon)
            if(isFileExit(saveUrl, res_size)):
                log('[000]__%s__already exist\n' % lat_lon)
            else:
                dl_size = 0
                chunk_size = 1024
                with open(saveUrl, "wb") as f:
                    for chunk in res.iter_content(chunk_size):
                        progress = round(dl_size/res_size*100, 2)
                        if chunk:
                            f.write(chunk)
                            print("\ralready dl {}%".format(progress), end="")
                            dl_size += chunk_size
                log('[200]__%s__End\n' % lat_lon)
                randomInt = random.randint(3, 6)  # 随机休眠3到6s
                time.sleep(randomInt)
        elif(res_code == 404):
            log('[404]__%s__is not exist\n' % lat_lon)
        else:
            log('[%s]__other_res_code__End\n' % res_code)
    except Exception as e:
        log('[XXX]__Exception__{}\n'.format(e))
        failLog(lat_lon)
    finally:
        res.close()


def getRandomUserAgent():
    MY_USER_AGENT = [
        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; AcooBrowser; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0; Acoo Browser; SLCC1; .NET CLR 2.0.50727; Media Center PC 5.0; .NET CLR 3.0.04506)",
        "Mozilla/4.0 (compatible; MSIE 7.0; AOL 9.5; AOLBuild 4337.35; Windows NT 5.1; .NET CLR 1.1.4322; .NET CLR 2.0.50727)",
        "Mozilla/5.0 (Windows; U; MSIE 9.0; Windows NT 9.0; en-US)",
        "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Win64; x64; Trident/5.0; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 2.0.50727; Media Center PC 6.0)",
        "Mozilla/5.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0; WOW64; Trident/4.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; .NET CLR 1.0.3705; .NET CLR 1.1.4322)",
        "Mozilla/4.0 (compatible; MSIE 7.0b; Windows NT 5.2; .NET CLR 1.1.4322; .NET CLR 2.0.50727; InfoPath.2; .NET CLR 3.0.04506.30)",
        "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN) AppleWebKit/523.15 (KHTML, like Gecko, Safari/419.3) Arora/0.3 (Change: 287 c9dfb30)",
        "Mozilla/5.0 (X11; U; Linux; en-US) AppleWebKit/527+ (KHTML, like Gecko, Safari/419.3) Arora/0.6",
        "Mozilla/5.0 (Windows; U; Windows NT 5.1; en-US; rv:1.8.1.2pre) Gecko/20070215 K-Ninja/2.1.1",
        "Mozilla/5.0 (Windows; U; Windows NT 5.1; zh-CN; rv:1.9) Gecko/20080705 Firefox/3.0 Kapiko/3.0",
        "Mozilla/5.0 (X11; Linux i686; U;) Gecko/20070322 Kazehakase/0.4.5",
        "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.8) Gecko Fedora/1.9.0.8-1.fc10 Kazehakase/0.5.6",
        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_3) AppleWebKit/535.20 (KHTML, like Gecko) Chrome/19.0.1036.7 Safari/535.20",
        "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; fr) Presto/2.9.168 Version/11.52",
        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/536.11 (KHTML, like Gecko) Chrome/20.0.1132.11 TaoBrowser/2.0 Safari/536.11",
        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.71 Safari/537.1 LBBROWSER",
        "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E; LBBROWSER)",
        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E; LBBROWSER)",
        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.84 Safari/535.11 LBBROWSER",
        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E)",
        "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E; QQBrowser/7.0.3698.400)",
        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E)",
        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SV1; QQDownload 732; .NET4.0C; .NET4.0E; 360SE)",
        "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; QQDownload 732; .NET4.0C; .NET4.0E)",
        "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.1; WOW64; Trident/5.0; SLCC2; .NET CLR 2.0.50727; .NET CLR 3.5.30729; .NET CLR 3.0.30729; Media Center PC 6.0; .NET4.0C; .NET4.0E)",
        "Mozilla/5.0 (Windows NT 5.1) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.89 Safari/537.1",
        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.1 (KHTML, like Gecko) Chrome/21.0.1180.89 Safari/537.1",
        "Mozilla/5.0 (iPad; U; CPU OS 4_2_1 like Mac OS X; zh-cn) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8C148 Safari/6533.18.5",
        "Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:2.0b13pre) Gecko/20110307 Firefox/4.0b13pre",
        "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:16.0) Gecko/20100101 Firefox/16.0",
        "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.11 (KHTML, like Gecko) Chrome/23.0.1271.64 Safari/537.11",
        "Mozilla/5.0 (X11; U; Linux x86_64; zh-CN; rv:1.9.2.10) Gecko/20100922 Ubuntu/10.10 (maverick) Firefox/3.6.10",
        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36",
    ]
    list_len = len(MY_USER_AGENT)
    randomInt = random.randint(0, list_len-1)
    return MY_USER_AGENT[randomInt]


def mkdir(path):
    folder = os.path.exists(path)
    if not folder:
        os.makedirs(path)
        print('file %s create OK' % (path))
    else:
        print('file %s already exist' % (path))


def createLogTxt(logDir):
    '''
    根据现在的时间生成日志文件
    logDir:日志文件目录
    return:logFile 返回log文件路径(目录加文件名)
    '''
    logTxtName = time.strftime("%Y-%m-%d %H_%M_%S", time.localtime())
    global logFile
    logFile = logDir+logTxtName+'.txt'
    log('__log init__\n')


def createFailLogTxt(logDir):
    logTxtName = time.strftime("%Y-%m-%d %H_%M_%S", time.localtime())
    global failLogFile
    failLogFile = logDir+logTxtName+'_fail.txt'
    failLog('__FialLog init__\n')


def log(info):
    '''
    记录日志
    logFile 日志文件路径
    '''
    print(info)
    timeStr = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
    # 以日期来创建log文件
    with open(logFile, "a") as f:
        f.write(timeStr+'_____'+info)


def failLog(info):
    '''
    记录失败的lat_lon
    '''
    with open(failLogFile, 'a')as f:
        f.write("%s," % info)


def getLatLonList(lat, latRange, lon, lonRange):
    """
    lat:"N"/"S"
    latRange:list,纬度范围
    lon:"W"/"E"
    lonRange:list,经度范围
    lat_lon_list:拼接部分组成的list
    """
    lat_lon_list = []
    for latNum in range(latRange[0], latRange[1]):
        for lonNum in range(lonRange[0], lonRange[1]):
            latStr = (str(latNum)).zfill(2)
            lonStr = (str(lonNum)).zfill(3)
            lat_lon = lat+latStr+lon+lonStr
            lat_lon_list.append(lat_lon)
    return lat_lon_list


def isFileExit(path, realSize):
    '''
    path:文件应该保存到的路径
    realSize:从服务端获取到的文件大小
    '''
    try:
        size = os.path.getsize(path)
        if(realSize == size):
            log('zip already dl, size is ok\n')
            return True
        else:
            log('zip already dl,size is shit,Downloading...\n')
            return False
    except FileNotFoundError:
        log('zip have not dl,Downloading...\n')
        return False


if __name__ == "__main__":
    # 文件夹初始化
    saveDir = './Download_Dem/'
    logDir = './Log/'
    failDir = './Fail/'
    mkdir(saveDir)
    mkdir(logDir)
    mkdir(failDir)
    createLogTxt(logDir)
    createFailLogTxt(failDir)

    # N00-N83,W01-W180
    list1 = getLatLonList('N', [0, 84], 'W', [1, 181])
    # N00-N83,E00-W179
    list2 = getLatLonList('N', [0, 84], 'E', [0, 180])
    # S01-S83,W01-W180
    list3 = getLatLonList('S', [1, 84], 'W', [1, 181])
    # S01-S83,E00-E179
    list4 = getLatLonList('S', [1, 84], 'E', [0, 180])

    list = list1+list2+list3+list4

    for lat_lon in list:
        url = "https://gdemdl.aster.jspacesystems.or.jp/download/Download_%s.zip" % lat_lon
        saveUrl = "%sDownload_%s.zip" % (saveDir, lat_lon)
        download(url, saveUrl)
        # thread = Thread(target=download, args=(url, saveUrl,))
        # thread.start()
